# codding: utf-8
import torch

class Config(object):
    def __init__(self):
        self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
        self.train_file_path = r'D:\AI_workplace\GitProjects\practice_pipline_relation_extraction\data_origin\train.txt'
        self.test_file_path = r'D:\AI_workplace\GitProjects\practice_pipline_relation_extraction\data_origin\test.txt'
        self.relation_file_path = r'D:\AI_workplace\GitProjects\practice_pipline_relation_extraction\data_origin\relation2id.txt'
        self.embedding_dim = 300
        self.hidden_dim = 128
        self.position_dim = 5
        self.dropout = 0.5
        self.epochs = 10
        self.batch_size = 4
        self.lr = 1e-3
        self.max_len = 70

